
import cv2


def main():
    num_gpu_devices = cv2.cuda.getCudaEnabledDeviceCount()
    print(num_gpu_devices)

    for i in range(num_gpu_devices):
        cv2.cuda.printCudaDeviceInfo(i)

    return

# cv2.gpu
#         cv::gpu::GpuMat dst, src;
#         src.upload(src_host);

#         cv::gpu::threshold(src, dst, 128.0, 255.0, CV_THRESH_BINARY);


if __name__ == '__main__':
    main()
